from pyecharts.charts import Bar,Timeline
from pyecharts.options import *
from pyecharts.globals import ThemeType
# 读取文件
f = open("//基础/数据可视化/动态柱形图/1960-2019全球GDP数据.csv", encoding="GB2312")
data_lines = f.readlines()
f.close()
# 删除第一行
data_lines.pop(0)
# 定义一个字典对象
data_dict ={}
for line in data_lines:
    year = int(line.split(",")[0])  # 年份
    country = line.split(",")[1]  # 国家
    gdp = float(line.split(",")[2])  # gdp数据
    # 判断字典有没有指定的key
    try:
        data_dict[year].append([country,gdp])
    except KeyError:
        data_dict[year] = []
        data_dict[year].append([country, gdp])
# 创建时间线对象
timeline = Timeline(
{"theme":ThemeType.LIGHT}
)
# 排序年份
sorted_year_list = sorted(data_dict.keys())
for year in sorted_year_list:
    data_dict[year].sort(key=lambda element:element[1],reverse=True)
    # 取出本年份前8名
    year_data = data_dict[year][0:8]
    x_data = []
    y_data = []
    for country_gdp in year_data:
        x_data.append(country_gdp[0])   # x轴添加国家
        y_data.append(country_gdp[1] / 100000000)   # y轴添加gdp
    # 构建柱形图
    bar =Bar()
    x_data.reverse()
    y_data.reverse()
    bar.add_xaxis(x_data)
    bar.add_yaxis("GDP（亿）",y_data,label_opts=LabelOpts(position="right"))
    # 翻转xy轴
    bar.reversal_axis()
    # 将每一年的bar对象添加到时间线
    bar.set_global_opts(
        title_opts=TitleOpts(title=f"{year}年全球前8GDP数据"),
        legend_opts={'position':'right'}# 设置每一年图表的标题
    )
    timeline.add(bar,str(year))
# 设置时间线自动播放
timeline.add_schema(
    play_interval=1000,         # 自动播放的时间间隔，单位毫秒
    is_timeline_show=True,   # 是否在自动播放时显示时间线
    is_auto_play=True,          # 是否自动播放
    is_loop_play=True          # 是否循环自动播放
)
# 绘图
timeline.render("1960-2019全球GDP前8国家.html")